home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Blender 2.49b / blender-2.49b-windows.exe / $_4_ / .blender / scripts / weightpaint_average.py < prev    next >
Text File  |  2009-08-31  |  4KB  |  121 lines
  1. #!BPY
  2. """
  3. Name: 'Vertex Groups Island Average'
  4. Blender: 243
  5. Group: 'WeightPaint'
  6. Tooltip: 'Average the vertex weights for each connected set of verts'
  7. """
  8.  
  9. # -------------------------------------------------------------------------- 
  10. # ***** BEGIN GPL LICENSE BLOCK ***** 
  11. # 
  12. # This program is free software; you can redistribute it and/or 
  13. # modify it under the terms of the GNU General Public License 
  14. # as published by the Free Software Foundation; either version 2 
  15. # of the License, or (at your option) any later version. 
  16. # 
  17. # This program is distributed in the hope that it will be useful, 
  18. # but WITHOUT ANY WARRANTY; without even the implied warranty of 
  19. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
  20. # GNU General Public License for more details. 
  21. # 
  22. # You should have received a copy of the GNU General Public License 
  23. # along with this program; if not, write to the Free Software Foundation, 
  24. # Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA. 
  25. # 
  26. # ***** END GPL LICENCE BLOCK ***** 
  27. # --------------------------------------------------------------------------
  28. import Blender
  29. from Blender import Scene, Mesh, Window, sys, Draw
  30. from BPyMesh import meshWeight2List, list2MeshWeight, mesh2linkedFaces
  31.  
  32. import BPyMessages
  33. import bpy
  34.  
  35. def faceGroups2VertSets(face_groups):
  36.     '''    Return the face groups as sets of vert indicies    '''
  37.     return [set([v.index for f in fg for v in f]) for fg in face_groups]
  38.  
  39.  
  40. def vgroup_average(ob_orig, me, sce, PREF_ALL_VGROUPS=True):
  41.     if not me.getVertGroupNames():
  42.         return
  43.     
  44.     weight_names, weight_list = meshWeight2List(me)
  45.     
  46.     weight_names_len = len(weight_names)
  47.     vgroup_dummy = [0.0] * weight_names_len
  48.     vgroup_range = range(weight_names_len)
  49.     
  50.     if not PREF_ALL_VGROUPS:
  51.         weight_active_index = weight_names.index(me.activeGroup)
  52.     
  53.     for vert_set in faceGroups2VertSets( mesh2linkedFaces(me) ):
  54.         if not vert_set:
  55.             continue
  56.         
  57.         
  58.         if PREF_ALL_VGROUPS:
  59.             # We need to average the vgroups
  60.             collected_group = vgroup_dummy[:]
  61.             for i in vert_set:
  62.                 vert_group = weight_list[i]            # get the original weight
  63.                 weight_list[i] = collected_group    # replace with the collected group
  64.                 
  65.                 for j in vgroup_range: # iter through the vgroups
  66.                     collected_group[j] += vert_group[j]
  67.             
  68.             for j in vgroup_range:
  69.                 collected_group[j] /= len(vert_set)
  70.         else:
  71.             # Active group only
  72.             vert_weight = 0.0
  73.             for i in vert_set:
  74.                 vert_weight += weight_list[i][weight_active_index]
  75.             
  76.             vert_weight /= len(vert_set)
  77.             
  78.             for i in vert_set:
  79.                 weight_list[i][weight_active_index] = vert_weight
  80.     
  81.     list2MeshWeight(me, weight_names, weight_list)
  82.  
  83. def main():
  84.     
  85.     # Gets the current scene, there can be many scenes in 1 blend file.
  86.     sce = bpy.data.scenes.active
  87.     
  88.     # Get the active object, there can only ever be 1
  89.     # and the active object is always the editmode object.
  90.     ob_act = sce.objects.active
  91.     
  92.     if not ob_act or ob_act.type != 'Mesh':
  93.         BPyMessages.Error_NoMeshActive()
  94.         return 
  95.     
  96.     # Saves the editmode state and go's out of 
  97.     # editmode if its enabled, we cant make
  98.     # changes to the mesh data while in editmode.
  99.     is_editmode = Window.EditMode()
  100.     Window.EditMode(0)
  101.     
  102.     PREF_ALL_VGROUPS = Draw.PupMenu("All Groups?%t|All Groups%x1|Active Group Only%x0")
  103.     if PREF_ALL_VGROUPS==-1:
  104.         return
  105.     
  106.     Window.WaitCursor(1)
  107.     me = ob_act.getData(mesh=1) # old NMesh api is default
  108.     t = sys.time()
  109.     
  110.     # Run the mesh editing function
  111.     vgroup_average(ob_act, me, sce, PREF_ALL_VGROUPS)
  112.     
  113.     # Timing the script is a good way to be aware on any speed hits when scripting
  114.     print 'Average VGroups in %.2f seconds' % (sys.time()-t)
  115.     Window.WaitCursor(0)
  116.     if is_editmode: Window.EditMode(1)
  117.     
  118.     
  119. # This lets you can import the script without running it
  120. if __name__ == '__main__':
  121.     main()